/**
 * Copyright (c) 2012, Brent Mucci
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without modification, 
 * are permitted provided that the following conditions are met:
 * 
 * 1. Redistributions of source code must retain the above copyright notice, 
 *    this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright notice, 
 *    this list of conditions and the following disclaimer in the documentation 
 *    and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR 
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON 
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, 
 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 **/

#include "mcl_nal_reader.hpp"

using namespace mcl;

NALReader::NALReader(std::istream &stream)
	: BitstreamBuffer(stream)
{

}

NALReader::~NALReader()
{
}

unsigned int NALReader::getNextNALSize()
{
	unsigned int 
		/*leading_zero_8bits,
		start_code_prefix_one_3bytes,
		zero_byte,
		trailing_zero_8bits,*/
		num_nal_bytes = 0;

	if(!super::more_data_in_byte_stream()){
		return 0;
	}

	while(super::next_bits(24) != 1 && super::next_bits(32) != 1){
		/*leading_zero_8bits = */super::read_bits(8);
	}

	if(super::next_bits(24) != 1){
		/*zero_byte = */super::read_bits(8);
	}

	/*start_code_prefix_one_3bytes = */super::read_bits(24);

	while(super::more_data_in_byte_stream() && super::next_bits(24) != 1 && super::next_bits(32) != 1)
	{
		num_nal_bytes++;
		/*trailing_zero_8bits = */super::read_bits(8);
	}

	return num_nal_bytes;
}